package org.szgb.console.sys.line.web;

import cn.hutool.http.HttpStatus;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.szgb.console.sys.line.entity.SysLine;
import org.szgb.console.sys.line.entity.SysLineLocation;
import org.szgb.console.sys.line.service.SysLineLocationService;
import org.szgb.console.sys.line.service.SysLineService;
import org.szgb.console.sys.user.entity.SysUser;
import org.szgb.console.utils.PageView;
import org.szgb.console.utils.TokenUtil;
import org.szgb.core.base.entity.GetIdDto;
import org.szgb.core.utils.AjaxJson;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@RestController
@RequestMapping(value = "/sys/line")
@Tag(name = "线路管理", description = "线路管理模块")
public class SysLineController {
    private final SysLineService sysLineService;
    private final SysLineLocationService sysLineLocationService;

    @Autowired
    public SysLineController(SysLineService sysLineService, SysLineLocationService sysLineLocationService) {
        this.sysLineService = sysLineService;
        this.sysLineLocationService = sysLineLocationService;
    }


    @Operation(summary = "获取线路信息", description = "根据查询参数获取线路信息，支持分页查询。使用id获取更详细的信息。")
    @PostMapping("/get")
    public AjaxJson get(@RequestBody(required = false) SysLine.SysLineFindDto findDto, HttpServletRequest request, HttpServletResponse response) {
        if (findDto == null) {
            findDto = new SysLine.SysLineFindDto();
        }
        // 创建一个PageView对象，用于分页显示
        PageView<SysLine.SysLineView> pageView = new PageView<>();
        SysLine sysLine = findDto.toSysLine();
        // 如果实体对象的ID不为空，则直接调用get方法获取单个实体对象
        if (sysLine != null && sysLine.getId() != null && !sysLine.getId().isEmpty()) {
            SysLine line = sysLineService.get(sysLine.getId());
            // 如果实体对象为空，则返回错误响应
            if (line == null) {
                return AjaxJson.error("数据不存在", HttpStatus.HTTP_NOT_FOUND, request, response);
            }
            // 返回成功的响应，并将实体对象放入响应中
            return AjaxJson.success(request, response).put("sysLine", line.toSysLineView());
        } else {
            // 设置Page对象的列表属性，包含根据实体对象查询到的数据
            pageView.setList(SysLine.SysLineView.getViewListFromList(sysLineService.findList(sysLine)));
            // 设置Page对象的总记录数属性，用于分页显示
            pageView.setCount(sysLineService.getCount(sysLine));
            // 创建一个成功的AjaxJson对象，并放入Page对象，以便于前端展示
            return AjaxJson.success(request, response).put("page", pageView);
        }
    }

    @Operation(summary = "获取线路树表信息", description = "根据查询参数获取线路树表信息，不填写id就是查询全部。")
    @PostMapping("/getTree")
    public AjaxJson getTree(@RequestBody(required = false) GetIdDto getIdDto, HttpServletRequest request, HttpServletResponse response) {
        if (getIdDto != null && getIdDto.getId() != null && !getIdDto.getId().isEmpty()) {
            SysLine sysLine = sysLineService.get(getIdDto.getId());
            SysLineLocation sysLineLocation = new SysLineLocation();
            sysLineLocation.setLine(sysLine);
            return AjaxJson.success(request, response).put("tree", new SysLine.SysLineTreeView(sysLine, SysLineLocation.SysLineLocationTreeItemView.getViewListFromList(sysLineLocationService.findList(sysLineLocation))));
        }
        List<SysLine.SysLineTreeView> sysLineTreeViewList = new ArrayList<>();
        List<SysLine> sysLineList = sysLineService.findList(new SysLine());
        List<SysLineLocation> locationList = sysLineLocationService.findList(new SysLineLocation());
        Map<String, List<SysLineLocation>> locationListMap = new HashMap<>();
        for (SysLine line : sysLineList) {
            locationListMap.put(line.getId(), new ArrayList<>());
        }
        locationList.forEach(location -> {
            locationListMap.get(location.getLine().getId()).add(location);
        });
        sysLineList.forEach(line -> {
            sysLineTreeViewList.add(new SysLine.SysLineTreeView(line, SysLineLocation.SysLineLocationTreeItemView.getViewListFromList(locationListMap.get(line.getId()))));
        });
        return AjaxJson.success(request, response).put("treeList", sysLineTreeViewList);
    }

    @PostMapping("/save")
    @Operation(summary = "保存线路", description = "新增或更改线路信息。")
    public AjaxJson save(@RequestBody SysLine.SysLineSaveDto sysLineSaveDto, HttpServletRequest
            request, HttpServletResponse response) {
        if (sysLineSaveDto == null) {
            return AjaxJson.error("参数错误", HttpStatus.HTTP_BAD_REQUEST, request, response);
        }
        SysLine sysLine = sysLineSaveDto.toSysLine();
        // 设置实体对象的更新者为当前用户
        sysLine.setUpdatedBy(new SysUser(TokenUtil.getUid(request)));
        // 调用服务层的save方法，尝试保存或更新实体对象，返回保存结果的Map
        sysLineService.save(sysLine);
        // 根据保存结果返回相应的AjaxJson对象
        return AjaxJson.success("保存成功", request, response).put("sysLine", sysLine.toSysLineView());
    }

    @DeleteMapping("/delete")
    @Operation(summary = "删除线路", description = "根据ID删除线路信息。", parameters = {@Parameter(name = "ids", description = "线路ID列表，多个ID用英文逗号分隔。")})
    public AjaxJson delete(@RequestParam String ids, HttpServletRequest request, HttpServletResponse response) {
        // 检查ID参数是否为空，为空则返回错误响应
        if (ids == null) {
            return AjaxJson.error("参数错误", HttpStatus.HTTP_BAD_REQUEST, request, response);
        }
        // 调用服务层方法，执行删除操作
        sysLineService.delete(ids);
        // 返回成功响应
        return AjaxJson.success(request, response);
    }
}
